const baseUrl = 'http://192.168.1.160'

const apiKey = 'ragflow-NiOTVhMDc0NmRmMjExZjA4YjE5YjYyND'
// c12dffe46c8c11f09c0fb6240119a399 工艺管理
// 9141d0586c8c11f08365b6240119a399 设备管理

const queryAllFileWidthDatasetId = async (dataset_id, type) => {
  const listApi = `${baseUrl}/api/v1/datasets/${dataset_id}/documents`
  const res = await fetch(`${listApi}?page=1&page_size=500`, {
    method: 'GET',
    headers: {
      'Content-Type': 'application/json',
      Authorization: 'Bearer ' + apiKey,
    },
  })
  if (res.ok) {
    const result = await res.json()
    const docs = result.data.docs || []
    if (docs.length > 0) {
      docs.forEach((doc) => {
        updateFileFiled(dataset_id, doc, type || '')
      })
    }
  }
}

const updateFileFiled = async (dataset_id, documentItem, type) => {
  console.log('documentItem', documentItem.name)
  const updateApi = `${baseUrl}/api/v1/datasets/${dataset_id}/documents/${documentItem.id}`
  const updateData = {
    meta_fields: {
      name: documentItem.name.substring(
        0,
        documentItem.name.lastIndexOf('.') - 1
      ),
      type: type,
    },
    chunk_method:
      documentItem.chunk_method === 'naive'
        ? 'manual'
        : documentItem.chunk_method,
  }
  console.log('updateData', updateData)
  const res = await fetch(`${updateApi}`, {
    method: 'PUT',
    headers: {
      'Content-Type': 'application/json',
      Authorization: 'Bearer ' + apiKey,
    },
    body: JSON.stringify(updateData),
  })
  if (res.ok) {
    const result = await res.json()
    console.log('data', JSON.stringify(result))
  }
}

// 00285dd46c8d11f0be47b6240119a399 管理制度
// c12dffe46c8c11f09c0fb6240119a399 工艺管理
// 9141d0586c8c11f08365b6240119a399 设备管理
// 6c6783cc6c8c11f0854db6240119a399 体系文件-设备管理
// d4ae534e6c8b11f0a814b6240119a399 体系文件-管理制度

const dataSetList = [
  {
    dataset_id: '00285dd46c8d11f0be47b6240119a399',
    type: '管理制度',
  },
  // {
  //   dataset_id: 'c12dffe46c8c11f09c0fb6240119a399',
  //   type: '工艺管理',
  // },
  {
    dataset_id: '9141d0586c8c11f08365b6240119a399',
    type: '设备管理',
  },
  {
    dataset_id: '6c6783cc6c8c11f0854db6240119a399',
    type: '体系文件-设备管理',
  },
  {
    dataset_id: 'd4ae534e6c8b11f0a814b6240119a399',
    type: '体系文件-管理制度',
  },
]

dataSetList.forEach((item) => {
  queryAllFileWidthDatasetId(item.dataset_id, item.type)
})
// queryAllFileWidthDatasetId('c12dffe46c8c11f09c0fb6240119a399')
